package util;

import info.monitorenter.cpdetector.io.*;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 * @author zhongwen
 * @description 文件处理的工具
 * @date 2021-12-22 20:05
 */
public class FileUtils {

    /**
     * 将字符串保存到文件中
     */
    public static void saveFile(File selectedFile, String text) {
        try {
            BufferedWriter writer = new BufferedWriter(new FileWriter(selectedFile));
            writer.write(text);
            writer.flush();
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 将字符串按指定编码保存到文件中
     */
    public static void saveFile(File selectedFile, String text, String charset) {
        try {
            BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(selectedFile), charset));
            writer.write(text);
            writer.flush();
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 从文件中读取字符串
     */
    public static String openFile(File selectedFile){
        StringBuilder sb = new StringBuilder();
        try {
            BufferedReader reader = new BufferedReader(new FileReader(selectedFile));
            String s;
            while ((s = reader.readLine()) != null){
                sb.append(s);
            }
            reader.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return sb.toString();
    }

    /**
     * 从文件中按行读取字符串
     */
    public static List<String> openFilesWithList(File selectedFile){
        List<String> arrs = new ArrayList<>();
        try {
            BufferedReader reader = new BufferedReader(new FileReader(selectedFile));
            String s;
            while ((s = reader.readLine()) != null){
                arrs.add(s);
            }
            reader.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return arrs;
    }

    /**
     * 按指定编码对文件解码，并按行读取字符串
     */
    public static List<String> openFilesWithList(File selectedFile, String charset){
        List<String> arrs = new ArrayList<>();
        try {
            BufferedReader reader = new BufferedReader(new InputStreamReader(new FileInputStream(selectedFile), charset));
            String s;
            while ((s = reader.readLine()) != null){
                arrs.add(s);
            }
            reader.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return arrs;
    }

    /**
     * 获取要保存文件的默认名
     */
    public static String getDefaultName(String text) {
        int index = text.indexOf("class");
        if(index == -1){
            return null;
        }
        int i = index + 6;
        while (i < text.length() && text.charAt(i) == ' ') {
            i++;
        }
        StringBuilder sb = new StringBuilder();
        while (i < text.length() && (Character.isDigit(text.charAt(i)) || Character.isLetter(text.charAt(i)))) {
            sb.append(text.charAt(i));
            i++;
        }
        return sb.toString();
    }
    /**
     * 利用第三方开源包cpdetector获取文件编码格式
     *
     * @param f 要判断文件编码格式的源文件的file对象
     * @author huanglei
     * @version 2012-7-12 14:05
     */
    public static String getFileEncode(File f) {
        /*
         * detector是探测器，它把探测任务交给具体的探测实现类的实例完成。
         * cpDetector内置了一些常用的探测实现类，这些探测实现类的实例可以通过add方法 加进来，如ParsingDetector、
         * JChardetFacade、ASCIIDetector、UnicodeDetector。
         * detector按照“谁最先返回非空的探测结果，就以该结果为准”的原则返回探测到的
         * 字符集编码。使用需要用到三个第三方JAR包：antlr.jar、chardet.jar和cpdetector.jar
         * cpDetector是基于统计学原理的，不保证完全正确。
         */
        CodepageDetectorProxy detector = CodepageDetectorProxy.getInstance();
        /*
         * ParsingDetector可用于检查HTML、XML等文件或字符流的编码,构造方法中的参数用于
         * 指示是否显示探测过程的详细信息，为false不显示。
         */
        detector.add(new ParsingDetector(false));
        /*
         * JChardetFacade封装了由Mozilla组织提供的JChardet，它可以完成大多数文件的编码
         * 测定。所以，一般有了这个探测器就可满足大多数项目的要求，如果你还不放心，可以
         * 再多加几个探测器，比如下面的ASCIIDetector、UnicodeDetector等。
         */
        detector.add(JChardetFacade.getInstance());// 用到antlr.jar、chardet.jar
        // ASCIIDetector用于ASCII编码测定
        detector.add(ASCIIDetector.getInstance());
        // UnicodeDetector用于Unicode家族编码的测定
        detector.add(UnicodeDetector.getInstance());
        java.nio.charset.Charset charset = null;
        try {
            charset = detector.detectCodepage(f.toURI().toURL());
        } catch (Exception ex) {
            ex.printStackTrace();
        }
        if (charset != null) {
            return charset.name();
        } else {
            return null;
        }
    }
}
